#-
# ==========================================================================
# Copyright (C) 1995 - 2006 Autodesk, Inc. and/or its licensors.  All 
# rights reserved.
#
# The coded instructions, statements, computer programs, and/or related 
# material (collectively the "Data") in these files contain unpublished 
# information proprietary to Autodesk, Inc. ("Autodesk") and/or its 
# licensors, which is protected by U.S. and Canadian federal copyright 
# law and by international treaties.
#
# The Data is provided for use exclusively by You. You have the right 
# to use, modify, and incorporate this Data into other products for 
# purposes authorized by the Autodesk software license agreement, 
# without fee.
#
# The copyright notices in the Software and this entire statement, 
# including the above license grant, this restriction and the 
# following disclaimer, must be included in all copies of the 
# Software, in whole or in part, and all derivative works of 
# the Software, unless such copies or derivative works are solely 
# in the form of machine-executable object code generated by a 
# source language processor.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND. 
# AUTODESK DOES NOT MAKE AND HEREBY DISCLAIMS ANY EXPRESS OR IMPLIED 
# WARRANTIES INCLUDING, BUT NOT LIMITED TO, THE WARRANTIES OF 
# NON-INFRINGEMENT, MERCHANTABILITY OR FITNESS FOR A PARTICULAR 
# PURPOSE, OR ARISING FROM A COURSE OF DEALING, USAGE, OR 
# TRADE PRACTICE. IN NO EVENT WILL AUTODESK AND/OR ITS LICENSORS 
# BE LIABLE FOR ANY LOST REVENUES, DATA, OR PROFITS, OR SPECIAL, 
# DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES, EVEN IF AUTODESK 
# AND/OR ITS LICENSORS HAS BEEN ADVISED OF THE POSSIBILITY 
# OR PROBABILITY OF SUCH DAMAGES.
#
# ==========================================================================
#+

# import maya.cmds
# maya.cmds.loadPlugin("blindDoubleDataCmd.py")
# maya.cmds.sphere()
# maya.cmds.spBlindDoubleData()

import sys
import maya.OpenMaya as OpenMaya
import maya.OpenMayaMPx as OpenMayaMPx

kPluginName = "spBlindDoubleData"
kPluginDataId = OpenMaya.MTypeId(0x87011)

#
fValueDictionary={}

# testing function
def printMsg(msg):
	print msg
	stream=OpenMaya.MStreamUtils.stdOutStream()
	OpenMaya.MStreamUtils.writeCharBuffer(stream,msg)

# data
class blindDoubleData(OpenMayaMPx.MPxData):
	def __init__(self):
		OpenMayaMPx.MPxData.__init__(self)
		self.__fValue = 0.0
		fValueDictionary[OpenMayaMPx.asHashable(self)]=self.__fValue

	def readASCII(self, args, lastParsedElement):
		try:
			if args.length() > 0:
				parsedIndex = OpenMaya.MScriptUtil.getUint(lastParsedElement)
				self.__fValue = args.asDouble( parsedIndex )
				parsedIndex += 1
				OpenMaya.MScriptUtil.setUint(lastParsedElement,parsedIndex)
				fValueDictionary[OpenMayaMPx.asHashable(self)]=self.__fValue
		except:
			sys.stderr.write("Failed to read ASCII value.")
			raise

	def readBinary(self, inStream, length):
		readParam = OpenMaya.MScriptUtil()
		readParam.createFromDouble(0.0)
		readPtr = readParam.asDoublePtr()
		OpenMaya.MStreamUtils.readDouble(inStream, readPtr, True )
		self.__fValue = OpenMaya.MScriptUtil.getDouble( readPtr )

	def writeASCII(self, out):
		try:
			OpenMaya.MStreamUtils.writeDouble(out, self.__fValue, False)
		except:
			sys.stderr.write("Failed to write ASCII value.")
			raise

	def writeBinary(self, out):
		try:
			OpenMaya.MStreamUtils.writeDouble(out, self.__fValue, True)
		except:
			sys.stderr.write("Failed to write binary value.")
			raise

	def copy(self, other):
		# Cannot convert other to self.  Use a dictionary
		# to hold the information we need.
		self.__fValue = fValueDictionary[OpenMayaMPx.asHashable(other)]

	def typeId(self):
		return kPluginDataId

	def name(self):
		return kPluginName

	def value(self):
		return self.__fValue

	def setValue(self, newVal):
		self.__fValue = newVal

# command
class blindDoubleDataCmd(OpenMayaMPx.MPxCommand):
	def __init__(self):
		OpenMayaMPx.MPxCommand.__init__(self)
		self.__iter = None

	def doIt(self, args):
		selList = OpenMaya.MSelectionList()
		OpenMaya.MGlobal.getActiveSelectionList(selList)
		self.__iter = OpenMaya.MItSelectionList(selList)
		self.redoIt()

	def redoIt(self):
		dependNode = OpenMaya.MObject() # Selected dependency node
		# show message and advance iterator
		def error(msg):
			sys.stderr.write(err)
			self.__iter.next()
			
		# Iterate over all selected dependency nodes
		#
		while not self.__iter.isDone():
			# Get the selected dependency node and create
			# a function set for it
			#
			try:
				self.__iter.getDependNode(dependNode)	
			except:
				error("Error getting the dependency node")
				continue

			try:
				fnDN = OpenMaya.MFnDependencyNode(dependNode)
			except:
				error("Error creating MFnDependencyNode")
				continue

			# Create a new attribute for our blind data
			#
			fnAttr = OpenMaya.MFnTypedAttribute()
			newAttr = fnAttr.create("blindDoubleData", "BDD", kPluginDataId)
			
			# Now add the new attribute to the current dependency node
			#
			fnDN.addAttribute(newAttr, OpenMaya.MFnDependencyNode.kLocalDynamicAttr)

			# Create a plug to set and retrive value off the node.
			#
			plug = OpenMaya.MPlug(dependNode, newAttr)

			# Instantiate blindDoubleData and set its value.
			#
			newData = OpenMayaMPx.asMPxPtr(blindDoubleData())
			newData.setValue(3.2)

			# Set the value for the plug.
			#
			plug.setMPxData(newData)

			# Now try to retrieve the value off the plug as an MObject.
			#
			try:
				sData = plug.asMObject()
			except:
				error("Error getting value off plug")
				continue
				
			# Convert the data back to MPxData.
			#
			pdFn = OpenMaya.MFnPluginData(sData)
			data = pdFn.constData()

			# Get the value.
			#
			if not data:
				error("Error: failed to retrieve data.")
				continue
				
			self.__iter.next()

	def undoIt(self):
		pass

	def isUndoable(self):
		return True


########################################################################


# Creators
def cmdCreator():
	return OpenMayaMPx.asMPxPtr(blindDoubleDataCmd())

def dataCreator():
	return OpenMayaMPx.asMPxPtr(blindDoubleData())

# Initialize the script plug-in
def initializePlugin(mobject):
	mplugin = OpenMayaMPx.MFnPlugin(mobject, "Autodesk", "1.0", "Any")
	try:
		mplugin.registerData(kPluginName, kPluginDataId, dataCreator)
	except:
		sys.stderr.write("Failed to register new data type: %s\n" % kPluginName)
		raise
		
	try:
		mplugin.registerCommand(kPluginName, cmdCreator)
	except:
		sys.stderr.write("Failed to register command: %s\n" % kPluginName)
		raise


# Uninitialize the script plug-in
def uninitializePlugin(mobject):
	mplugin = OpenMayaMPx.MFnPlugin(mobject)
	try:
		mplugin.deregisterCommand(kPluginName)
	except:
		sys.stderr.write("Failed to unregister command: %s\n" % kPluginName)
		raise

	try:
		mplugin.deregisterData(kPluginDataId)
	except:
		sys.stderr.write("Failed to unregister data type: %s\n" % kPluginName)
		raise

